package leetcode.list;

/**
 * 19. 删除链表的倒数第 N 个结点
 * 给你一个链表，删除链表的倒数第 n 个结点，并且返回链表的头结点。
 */
public class _19 {
    public static void main(String[] args) {
        ListNode node1 = new ListNode(1);
//        ListNode node2 = new ListNode(2);
//        ListNode node3 = new ListNode(3);
//        ListNode node4 = new ListNode(4);

//        node1.next = node2;
//        node2.next = node3;
//        node3.next = node4;


        ListNode listNode = removeNthFromEnd(node1, 1);
        while (listNode != null) {
            System.out.println("listNode.val = " + listNode.val);
            listNode = listNode.next;
        }
    }

    public static ListNode removeNthFromEnd(ListNode head, int n) {
        // 左指针
        ListNode left = new ListNode(-1, head);

        // 右指针
        ListNode right = head;

        // 计数器
        int index = 1;

        while (right.next != null) {
            index++;
            right = right.next;

            if (index > n) {
                left = left.next;
            }
        }

        if (index >= n) {
            // 要删除头节点的话 直接
            if (left.next == head) {
                left.next = left.next.next;
                return left.next;
            } else {
                left.next = left.next.next;
                return head;
            }
        }

        return head;
    }


    public static class ListNode {
        int val;
        ListNode next;

        ListNode() {
        }

        ListNode(int val) {
            this.val = val;
        }

        ListNode(int val, ListNode next) {
            this.val = val;
            this.next = next;
        }
    }

}
